#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
数据库重置脚本
用于清空数据库并重新创建表结构
"""

import os
import sqlite3
from datetime import datetime

def reset_database():
    """重置数据库"""
    db_path = 'instance/student_forms.db'
    
    # 删除现有数据库文件
    if os.path.exists(db_path):
        os.remove(db_path)
        print(f"已删除现有数据库文件: {db_path}")
    
    # 确保instance目录存在
    os.makedirs('instance', exist_ok=True)
    
    # 创建新的数据库和表
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    
    # 创建学员表单表
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS student_forms (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            phone TEXT NOT NULL,
            age INTEGER NOT NULL,
            grade INTEGER NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    
    conn.commit()
    conn.close()
    
    print(f"数据库重置完成: {db_path}")
    print("新数据库已创建，包含空的student_forms表")

if __name__ == '__main__':
    reset_database()